package com.zty.util;

import lombok.extern.slf4j.Slf4j;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.security.MessageDigest;
import java.util.*;

/**
 * 小滴课堂,愿景：让技术不再难学
 *
 * @Description
 * @Author 二当家小D
 * @Remark 有问题直接联系我，源码-笔记-技术交流群
 * @Version 1.0
 **/

@Slf4j
public class CommonUtil {
    /**
     * 获取ip
     *
     * @param request
     * @return
     */
    public static String getIpAddr(HttpServletRequest request) {
        String ipAddress = null;
        try {
            // 尝试从HTTP头信息"x-forwarded-for"中获取IP地址
            // 这个头信息通常由代理服务器添加，用于标识原始请求的IP地址
            ipAddress = request.getHeader("x-forwarded-for");

            // 检查获取到的IP地址是否为空，长度是否为0，或者是否为字符串"unknown"（忽略大小写）
            // 如果是以上任一情况，则认为没有有效地从"x-forwarded-for"头中获取到IP地址
            if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {
                // 尝试从HTTP头信息"Proxy-Client-IP"中获取IP地址
                ipAddress = request.getHeader("Proxy-Client-IP");
            }

            // 继续检查获取到的IP地址是否有效，逻辑同上
            if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {
                // 尝试从HTTP头信息"WL-Proxy-Client-IP"中获取IP地址
                // 这个头信息是由WebLogic服务器添加的
                ipAddress = request.getHeader("WL-Proxy-Client-IP");
            }

            // 如果到目前为止，仍未获取到有效的IP地址
            if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {
                // 获取请求的远程IP地址，这通常是直接连接服务器的客户端IP地址
                ipAddress = request.getRemoteAddr();

                // 如果远程IP地址是"127.0.0.1"，即本机地址
                if (ipAddress.equals("127.0.0.1")) {
                    // 尝试通过网络接口获取本机的IP地址
                    InetAddress inet = null;
                    try {
                        inet = InetAddress.getLocalHost();
                    } catch (UnknownHostException e) {
                        // 如果获取失败，则打印堆栈跟踪信息
                        e.printStackTrace();
                    }
                    // 获取并设置本机的IP地址
                    ipAddress = inet.getHostAddress();
                }
            }

            // 对于经过多个代理的情况，IP地址可能包含多个值，由逗号分隔
            // 只取第一个IP地址作为客户端的真实IP
            if (ipAddress != null && ipAddress.length() > 15) {
                // 如果IP地址字符串的长度大于15，说明可能包含多个IP地址
                if (ipAddress.indexOf(",") > 0) {
                    // 只取第一个IP地址
                    ipAddress = ipAddress.substring(0, ipAddress.indexOf(","));
                }
            }

        } catch (Exception e) {
            ipAddress = "";
        }
        return ipAddress;
    }


    /**
     * 获取全部请求头
     *
     * @param request
     * @return
     */
    public static Map<String, String> getAllRequestHeader(HttpServletRequest request) {
        Enumeration<String> headerNames = request.getHeaderNames();
        Map<String, String> map = new HashMap<>();
        while (headerNames.hasMoreElements()) {
            String key = (String) headerNames.nextElement();
            //根据名称获取请求头的值
            String value = request.getHeader(key);
            map.put(key, value);
        }

        return map;
    }


    /**
     * MD5加密
     *
     * @param data
     * @return
     */
    public static String MD5(String data) {
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");
            byte[] array = md.digest(data.getBytes("UTF-8"));
            StringBuilder sb = new StringBuilder();
            for (byte item : array) {
                sb.append(Integer.toHexString((item & 0xFF) | 0x100).substring(1, 3));
            }

            return sb.toString().toUpperCase();
        } catch (Exception exception) {
        }
        return null;

    }


    /**
     * 获取验证码随机数
     *
     * @param length
     * @return
     */
    public static String getRandomCode(int length) {

        String sources = "0123456789";
        Random random = new Random();
        StringBuilder sb = new StringBuilder();
        for (int j = 0; j < length; j++) {
            sb.append(sources.charAt(random.nextInt(9)));
        }
        return sb.toString();
    }


    /**
     * 获取当前时间戳
     *
     * @return
     */
    public static long getCurrentTimestamp() {
        return System.currentTimeMillis();
    }


    /**
     * 生成uuid
     *
     * @return
     */
    public static String generateUUID() {
        return UUID.randomUUID().toString().replaceAll("-", "").substring(0, 32);
    }

    /**
     * 获取随机长度的串
     *
     * @param length
     * @return
     */
    private static final String ALL_CHAR_NUM = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";

    public static String getStringNumRandom(int length) {
        //生成随机数字和字母,
        Random random = new Random();
        StringBuilder saltString = new StringBuilder(length);
        for (int i = 1; i <= length; ++i) {
            saltString.append(ALL_CHAR_NUM.charAt(random.nextInt(ALL_CHAR_NUM.length())));
        }
        return saltString.toString();
    }


    /**
     * 响应json数据给前端
     *
     * @param response
     * @param obj
     */
    public static void sendJsonMessage(HttpServletResponse response, Object obj) {

        // 设置HTTP响应的内容类型为JSON，字符编码为UTF-8
        // 这告诉前端，响应体的内容是JSON格式的数据，且编码为UTF-8
        response.setContentType("application/json; charset=utf-8");

        // 使用try-with-resources语句来自动管理PrintWriter的资源
        // 这种结构可以确保在try块执行完毕后，PrintWriter资源将被自动关闭
        try (PrintWriter writer = response.getWriter()) {
            // 调用JsonUtil的obj2Json方法将Java对象转换为JSON格式的字符串
            // 然后使用PrintWriter将这个字符串打印到HTTP响应中
            writer.print(JsonUtil.obj2Json(obj));

            // 刷新缓冲区，强制任何缓冲的输出数据被写出到客户端
            // 这是为了确保JSON数据被完全发送出去
            response.flushBuffer();

        } catch (IOException e) {
            // 如果在获取PrintWriter、转换对象为JSON或者写出数据时遇到IO异常
            // 则捕获异常，并使用日志记录警告信息
            // 这里的{}是占位符，用于在日志信息中插入异常信息
            log.warn("响应json数据给前端异常:{}", e);
        }
    }

}